package cn.tedu.v6;

import org.w3c.dom.UserDataHandler;

import java.io.*;
import java.net.*;
import java.util.*;

/**
 * Version 6 在线消息列表
 */
public class Server {
    private ServerSocket serverSocket;
    private static final Integer PORT=6666;
    private List<PrintWriter> allOut=new ArrayList<>();
    //1.私有化一个集合保存所有用户名---OnlineUsers
    private Set<String> OnlineUsers=new HashSet<>();
    private Map<String,String> userDataBase;
    public Server(){
        try {
            serverSocket=new ServerSocket(PORT);
            userDataBase=new HashMap<>();
            userDataBase.put("空桑少主","123456");
            userDataBase.put("鹄羹","654321");
            userDataBase.put("玻璃鱿鱼","123");
        } catch (BindException e){
            System.out.println("端口号重复申请，建议换个端口号");
        } catch (IOException e) {
            System.out.println("申请端口号失败...");
        }
    }

    public void start()  {
        try {
            while (true){
            System.out.println("-------等待客户端链接------");
            Socket socket = serverSocket.accept();

            new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开");
        } finally {
            if (serverSocket!=null&&!serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }
        }
    }

    public static void main(String[] args)  {
        new Server().start();
    }
    private synchronized void broadcastMessage(String message){
        for (PrintWriter pw:allOut){
            pw.println("消息广播"+message);
        }
    }
    //2.封装一个方法:同步转发给所有客户端消息: “在线用户: ”+OnlineUsers
    //updateOnlineUsers  更新在线人名单的方法: “在线用户: ”+OnlineUsers
    //字符串: String userList="在线用户"+String.join("、",OnlineUsers)
   /* private synchronized void OnlineUsers(String message){
        for (String str:OnlineUsers){
            str.equals("在线用户: "+OnlineUsers);
            String userList="在线用户"+String.join( "、", OnlineUsers);
        }
    }*/
   // String userList="在线用户"+String.join("、",OnlineUsers);
    private synchronized void updateOnlineUsers(String message){
        String userList="在线用户"+String.join("、",OnlineUsers);
        for (PrintWriter p:allOut){
            p.println("在线用户: "+userList);
            //String userList="在线用户"+String.join("、",OnlineUsers);
        }
    }
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;
        private String username;
        private String password;
        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();

            try {
                 br=new BufferedReader(new InputStreamReader
                        (socket.getInputStream(),"UTF8"));
                 pw = new PrintWriter(new BufferedWriter(new OutputStreamWriter
                        (socket.getOutputStream(),
                                "UTF8")), true);
                synchronized (allOut) {
                    allOut.add(pw);
                }

            }  catch (IOException e) {
                System.out.println("客户端断开通讯~");
            }
        }

        private boolean Login(String user,String pass){
            return userDataBase.containsKey(user)
                    &&userDataBase.get(user).equals(pass);
        }

        private void handlerChat(){
            new Thread(()->{
                try{
                    String message;
                    String userList;
                    while ((message=br.readLine())!=null){
                        System.out.println("客户端说: "+message);
                        System.out.println("服务器端请输入: ");
                        if(message.equalsIgnoreCase("exit")){
                            pw.println(username+"退出聊天室！");
                            break;
                        }
                        broadcastMessage(message);
                        pw.println("请输入消息 (输入 exit 退出):");
                    }
                }catch (IOException e){
                    System.out.println("客户端读入失败!");
                }
            }).start();
            Scanner input=new Scanner(System.in);
            while(true){
                System.out.println(ip+"服务器请输入: ");
                String say=input.nextLine();
                if (say.equalsIgnoreCase("exit")){
                    System.out.println("服务器端即将关闭...");
                    break;
                }
                pw.println("客户端说: "+say);
            }
        }
        public void run() {
            try {
            pw.println("请输入用户名: ");
            username=br.readLine();
            pw.println("请输入密码: ");
            password=br.readLine();
            if (Login(username,password)){
                pw.println("登录成功，欢迎进入聊天室");
                //3.把所有用户添加至集合中【同步块: 登录人员】
                //3.1 转发在线用户方法
              /*  synchronized( OnlineUsers=updateOnlineUsers(username)){
                }*/
               // broadcastMessage("在线用户: "+ip);
                synchronized (OnlineUsers){
                    OnlineUsers.add(username);
                }
                updateOnlineUsers(username);
                handlerChat();
            }else {
                pw.println("登录失败，用户名或者密码错误!");
                socket.close();
            }
            } catch (NullPointerException e) {
                System.out.println("IO流传输对象，有对象为null,请立马核对。"
                        +"建议: 查看try块中对象可能赋值为null");
            } catch (IOException e){
                System.out.println("服务器端断开连接");
            }finally {
                //4.一个客户端断开链接，移除当前集合【同步】
                synchronized (OnlineUsers){
                    //提示xxx退出聊天室
                    System.out.println(username+"退出聊天室");
                    //if判断用户名是否正确
                    if (OnlineUsers.contains(username)){
                        OnlineUsers.remove(username);
                    }
                }
                allOut.remove(pw);
                synchronized (allOut) {
                    allOut.remove(pw);
                }
                if (br!=null) {
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
                }
                if (pw!=null) pw.close();
                if (socket!=null&&!socket.isClosed()){
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
                }
            }
        }
    }
}
